Learn R Programming

MXM (version 0.9.5)

Generate random folds for cross-validation: Generate random folds for cross-validation

Description

Random folds for use in a cross validation are generated. There is the option for stratified splitting as well.

Usage

generatefolds(target, nfolds = 10, stratified = TRUE, seed = FALSE)

Arguments

target
A vector with some data, either continuous or categorical.
nfolds
The number of folds to produce.
stratified
A boolean variable specifying whether stratified random (TRUE) or simple random (FALSE) sampling is to be used when producing the folds.
seed
A boolean variable. If set to TRUE, the folds will always be the same.

Value

A list with nfolds elements where each elements is a fold containing the indices of the data.

Details

I was inspired by the sam command in the package TunePareto in order to do the stratified version.

See Also

cv.ses

Examples

Run this code
a <- generatefolds(iris[, 5], nfolds = 5, stratified = TRUE)
table(iris[a[[1]], 5])  ## 10 values from each group

Run the code above in your browser using DataLab